from keras import Input
from keras.layers import BatchNormalizationV2

from .base_layer import BaseLayer


class BatchNormalizationLayer(BaseLayer):
    def transfer(self, model_list_line: list, inputs: Input, models: list = None):
        return BatchNormalizationV2()(inputs)